package com.wmh.baseservice.common.job.quartz;

import org.quartz.*;
import org.springframework.stereotype.Component;

/**
 * {@link org.quartz.Scheduler} 的管理器，负责创建任务
 * <p>
 * 另外，jobHandlerName 对应到 Spring Bean 的名字，直接调用
 *
 * @author 谢骏
 * @date 2023/8/23 11:19
 **/
@Component
public class SchedulerManager {

    private final Scheduler scheduler;

    public SchedulerManager(Scheduler scheduler) {
        this.scheduler = scheduler;
    }

    /**
     * 添加 Job 到 Quartz 中
     *
     * @param jobId           任务编号
     * @param jobHandlerName  任务处理器的名字
     * @param jobHandlerParam 任务处理器的参数
     * @param cronExpression  CRON 表达式
     * @param retryCount      重试次数
     * @param retryInterval   重试间隔
     * @throws SchedulerException 添加异常
     */
    public void addJob(Long jobId, String jobHandlerName, String jobHandlerParam, String cronExpression,
                       Integer retryCount, Integer retryInterval)
            throws SchedulerException {
        // 创建 JobDetail 对象
        JobDetail jobDetail = JobBuilder.newJob(JobHandlerInvoker.class)
                .usingJobData(JobDataKeyEnum.JOB_ID.name(), jobId)
                .usingJobData(JobDataKeyEnum.JOB_HANDLER_NAME.name(), jobHandlerName)
                .withIdentity(String.valueOf(jobId)).build();
        // 创建 Trigger 对象
        Trigger trigger = this.buildTrigger(jobId, jobHandlerName, jobHandlerParam, cronExpression, retryCount, retryInterval);
        // 新增调度
        scheduler.scheduleJob(jobDetail, trigger);
    }

    /**
     * 更新 Job 到 Quartz
     *
     * @param jobHandlerName  任务处理器的名字
     * @param jobHandlerParam 任务处理器的参数
     * @param cronExpression  CRON 表达式
     * @param retryCount      重试次数
     * @param retryInterval   重试间隔
     * @throws SchedulerException 更新异常
     */
    public void updateJob(Long jobId, String jobHandlerName, String jobHandlerParam, String cronExpression,
                          Integer retryCount, Integer retryInterval)
            throws SchedulerException {
        // 创建新 Trigger 对象
        Trigger newTrigger = this.buildTrigger(jobId, jobHandlerName, jobHandlerParam, cronExpression, retryCount, retryInterval);
        // 修改调度
        scheduler.rescheduleJob(new TriggerKey(String.valueOf(jobId)), newTrigger);
    }

    /**
     * 删除 Quartz 中的 Job
     *
     * @param jobIb 任务唯一编号
     * @throws SchedulerException 删除异常
     */
    public void deleteJob(Long jobIb) throws SchedulerException {
        scheduler.deleteJob(new JobKey(String.valueOf(jobIb)));
    }

    /**
     * 暂停 Quartz 中的 Job
     *
     * @param jobIb 任务唯一编号
     * @throws SchedulerException 暂停异常
     */
    public void pauseJob(Long jobIb) throws SchedulerException {
        scheduler.pauseJob(new JobKey(String.valueOf(jobIb)));
    }

    /**
     * 启动 Quartz 中的 Job
     *
     * @param jobIb 任务唯一编号
     * @throws SchedulerException 启动异常
     */
    public void resumeJob(Long jobIb) throws SchedulerException {
        scheduler.resumeJob(new JobKey(String.valueOf(jobIb)));
        scheduler.resumeTrigger(new TriggerKey(String.valueOf(jobIb)));
    }

    /**
     * 立即触发一次 Quartz 中的 Job
     *
     * @param jobId           任务编号
     * @param jobHandlerName  任务处理器的名字
     * @param jobHandlerParam 任务处理器的参数
     * @throws SchedulerException 触发异常
     */
    public void triggerJob(Long jobId, String jobHandlerName, String jobHandlerParam)
            throws SchedulerException {
        // 无需重试，所以不设置 retryCount 和 retryInterval
        JobDataMap data = new JobDataMap();
        data.put(JobDataKeyEnum.JOB_ID.name(), jobId);
        data.put(JobDataKeyEnum.JOB_HANDLER_NAME.name(), jobHandlerName);
        data.put(JobDataKeyEnum.JOB_HANDLER_PARAM.name(), jobHandlerParam);
        // 触发任务
        scheduler.triggerJob(new JobKey(String.valueOf(jobId)), data);
    }

    private Trigger buildTrigger(Long jobId, String jobHandlerName, String jobHandlerParam, String cronExpression,
                                 Integer retryCount, Integer retryInterval) {
        return TriggerBuilder.newTrigger()
                .withIdentity(String.valueOf(jobId))
                .withSchedule(CronScheduleBuilder.cronSchedule(cronExpression))
                .usingJobData(JobDataKeyEnum.JOB_HANDLER_PARAM.name(), jobHandlerParam)
                .usingJobData(JobDataKeyEnum.JOB_RETRY_COUNT.name(), retryCount)
                .usingJobData(JobDataKeyEnum.JOB_RETRY_INTERVAL.name(), retryInterval)
                .build();
    }
}
